import fitz  # PyMuPDF库
import os
import time  # 导入time模块

# 指定包含PDF文件的文件夹路径
pdf_folder_path = r'/Users/moss/Desktop/output'
# 定义图片保存的基础路径
base_save_path = r'/Users/moss/Desktop/output'

# 使用os库列出文件夹中所有以.pdf结尾的文件
pdf_files = [f for f in os.listdir(pdf_folder_path) if f.endswith('.pdf')]

# 记录总的开始时间
start_time = time.time()

# 遍历每个PDF文件
for pdf_file in pdf_files:
    # 记录每个PDF文件的开始时间
    file_start_time = time.time()
    
    # 拼接完整的PDF文件路径
    pdf_path = os.path.join(pdf_folder_path, pdf_file)
    
    # 根据PDF文件名创建子文件夹，用于存放转换后的图片
    save_path = os.path.join(base_save_path, pdf_file[:-4])  # 移除.pdf后缀
    if not os.path.exists(save_path):
        os.makedirs(save_path)  # 如果子文件夹不存在，则创建

    # 使用fitz打开PDF文件
    doc = fitz.open(pdf_path)

    # 遍历PDF中的每一页
    for page_number in range(len(doc)):
        # 加载当前页
        page = doc.load_page(page_number)

        # 将当前页转换为图片（pix对象），设置DPI为300以获得较高的图片质量
        pix = page.get_pixmap(dpi=300)

        # 定义当前页图片的保存路径和文件名
        image_path = os.path.join(save_path, f'page_{page_number + 1}.png')

        # 保存图片到指定路径
        pix.save(image_path)

    # 处理完当前PDF文件后关闭文档对象
    doc.close()

    # 记录每个PDF文件的结束时间并计算耗时
    file_end_time = time.time()
    file_duration = file_end_time - file_start_time
    print(f"{pdf_file} 转换完成。耗时: {file_duration:.2f} 秒")

# 记录总的结束时间并计算总耗时
end_time = time.time()
total_duration = end_time - start_time
print(f"所有PDF文件转换完成。总耗时: {total_duration:.2f} 秒")
